from django.db import migrations

def insert_top_orgs(apps, schema_editor):
    TopOrg = apps.get_model('trend', 'TopOrg')
    
    orgs_data = [
        {'org': '华为技术有限公司', 'count': 2500},
        {'org': '腾讯科技(深圳)有限公司', 'count': 1800},
        {'org': '阿里巴巴(中国)有限公司', 'count': 1600},
        {'org': '百度在线网络技术(北京)有限公司', 'count': 1400},
        {'org': '字节跳动有限公司', 'count': 1200},
        {'org': '小米科技有限责任公司', 'count': 1000},
        {'org': '京东科技有限公司', 'count': 800},
        {'org': '网易(杭州)网络有限公司', 'count': 600},
        {'org': '美团点评科技有限公司', 'count': 500},
        {'org': '滴滴出行科技有限公司', 'count': 400},
    ]
    
    for data in orgs_data:
        TopOrg.objects.create(**data)

def reverse_top_orgs(apps, schema_editor):
    TopOrg = apps.get_model('trend', 'TopOrg')
    TopOrg.objects.all().delete()

class Migration(migrations.Migration):
    dependencies = [
        ('trend', '04_ins_word_pairs'),  # 依赖于04_ins_word_pairs
    ]

    operations = [
        migrations.RunPython(insert_top_orgs, reverse_top_orgs),
    ] 